diff --git a/src/arcaflow_plugin_sdk/test_schema.py b/src/arcaflow_plugin_sdk/test_schema.py
index 0d7bc2a..8d77019 100644
--- a/src/arcaflow_plugin_sdk/test_schema.py
+++ b/src/arcaflow_plugin_sdk/test_schema.py
@@ -596,7 +596,7 @@ def setUp(self):
                     InlineInt,
                     {
                         discriminator_field_name: PropertyType(
-                            schema.StringType(),
+                            schema.IntType(),
                         ),
                         "msg": PropertyType(schema.StringType()),
                         "code": PropertyType(schema.IntType()),
@@ -606,7 +606,7 @@ def setUp(self):
                     InlineInt2,
                     {
                         discriminator_field_name: PropertyType(
-                            schema.StringType(),
+                            schema.IntType(),
                         ),
                         "msg2": PropertyType(schema.StringType()),
                     },