diff --git a/flake_data_nodes.py b/flake_data_nodes.py
index f3d6553..93a6dff 100644
--- a/flake_data_nodes.py
+++ b/flake_data_nodes.py
@@ -24,8 +24,8 @@ _ALL_PINS = [
     ("height", "INT"),
     ("steps", "INT"),
     ("cfg", "FLOAT"),
-    ("sampler_name", _SAMPLER_TYPES[0]),
-    ("scheduler", _SAMPLER_TYPES[1]),
+    ("sampler_name", "SAMPLER"),
+    ("scheduler", "SCHEDULER"),
 ]