diff --git a/src/arcaflow_plugin_sdk/test_schema.py b/src/arcaflow_plugin_sdk/test_schema.py
index 7a9441f..8d77019 100644
--- a/src/arcaflow_plugin_sdk/test_schema.py
+++ b/src/arcaflow_plugin_sdk/test_schema.py
@@ -12,7 +12,7 @@
     BadArgumentException,
     ConstraintException,
     PropertyType,
-    SchemaBuildException, step_object_constructor_param,
+    SchemaBuildException,
 )
 
 # default discriminator field name used by the OneOfType