diff --git a/stablehlo/integrations/python/CMakeLists.txt b/stablehlo/integrations/python/CMakeLists.txt
index d231cb31e6..fac75773a1 100644
--- a/stablehlo/integrations/python/CMakeLists.txt
+++ b/stablehlo/integrations/python/CMakeLists.txt
@@ -65,15 +65,6 @@ declare_mlir_python_sources(StablehloToSavedModelPythonSources
     stablehlo/savedmodel/stablehlo_to_tf_saved_model.py
 )
 
-declare_mlir_python_sources(StablehloTestdataGeneratorPythonSources
-  ADD_TO_PARENT StablehloPythonSources
-  ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}"
-  SOURCES
-    stablehlo/testdata_generator/testdata_execution_utils.py
-    stablehlo/testdata_generator/testdata_generator_lib.py
-    stablehlo/testdata_generator/testdata_processor.py
-)
-
 declare_mlir_python_sources(VhloPythonSources)
 declare_mlir_python_sources(VhloPythonSources.Dialects
   ADD_TO_PARENT VhloPythonSources
diff --git a/stablehlo/integrations/python/tests/check.py b/stablehlo/integrations/python/tests/check.py
index 964a503aca..70c03adbc2 100644
--- a/stablehlo/integrations/python/tests/check.py
+++ b/stablehlo/integrations/python/tests/check.py
@@ -23,8 +23,8 @@
 
 def run(f):
   with ir.Context() as context:
-    check.register_dialect(context)
-    stablehlo_dialect.register_dialect(ctx)
+    check_dialect.register_dialect(context)
+    stablehlo_dialect.register_dialect(context)
     f()
   return f