1 files changed, 9 insertions, 2 deletions
diff --git a/Scripts/Fold/Editor/FoldGraphView.cs b/Scripts/Fold/Editor/FoldGraphView.cs
index f77e97b..b8911c1 100644
--- a/Scripts/Fold/Editor/FoldGraphView.cs
+++ b/Scripts/Fold/Editor/FoldGraphView.cs
@@ -11,8 +11,15 @@ public class FoldGraphView : BaseGraphView
 
     public override IEnumerable<(string path, Type type)> FilterCreateNodeMenuEntries()
     {
-        return NodeProvider.GetNodeMenuEntries(graph)
-            .Where(entry => typeof(BaseFoldNode).IsAssignableFrom(entry.type));
+      return NodeProvider.GetNodeMenuEntries(graph)
+        .Where(entry =>
+            typeof(BaseFoldNode).IsAssignableFrom(entry.type) ||
+            typeof(KeyframeNode).IsAssignableFrom(entry.type) ||
+            // Whitelist a few built in nodes that we use.
+            typeof(FloatNode).IsAssignableFrom(entry.type) ||
+            typeof(GameObjectNode).IsAssignableFrom(entry.type) ||
+            typeof(VectorNode).IsAssignableFrom(entry.type)
+            );
     }
 }