diff --git a/tests/env.py b/tests/env.py
index 9de53999b..d234c425d 100644
--- a/tests/env.py
+++ b/tests/env.py
@@ -16,7 +16,7 @@ GRAALPY = sys.implementation.name == "graalpy"
 _graalpy_version = (
     sys.modules["__graalpython__"].get_graalvm_version() if GRAALPY else "0.0.0"
 )
-GRAALPY_VERSION = tuple(int(t) for t in _graalpy_version.split(".")[:3])
+GRAALPY_VERSION = tuple(int(t) for t in _graalpy_version.split("-")[0].split(".")[:3])
 PY_GIL_DISABLED = bool(sysconfig.get_config_var("Py_GIL_DISABLED"))
 
 
diff --git a/tests/test_callbacks.py b/tests/test_callbacks.py
index c43a9290a..9eab10ccb 100644
--- a/tests/test_callbacks.py
+++ b/tests/test_callbacks.py
@@ -223,7 +223,6 @@ def test_custom_func():
     assert m.roundtrip(m.custom_function)(4) == 36
 
 
-@pytest.mark.skipif("env.GRAALPY", reason="TODO debug segfault")
 def test_custom_func2():
     assert m.custom_function2(3) == 27
     assert m.roundtrip(m.custom_function2)(3) == 27