diff --git a/src/cb_dasIMGUI_NODE_EDITOR.h b/src/cb_dasIMGUI_NODE_EDITOR.h
index f089c59..48dfe67 100644
--- a/src/cb_dasIMGUI_NODE_EDITOR.h
+++ b/src/cb_dasIMGUI_NODE_EDITOR.h
@@ -93,7 +93,7 @@ struct imgui_node_editorTempFn {
 
         bool anyString = false;
         for ( auto &arg : fn->arguments ) {
-            if ( arg->type->constant && arg->type->ref && arg->type->dim.size() == 0 ) {
+            if ( arg->type->constant && arg->type->ref && !arg->type->isArray() ) {
                 if ( arg->type->baseType == Type::tFloat2 || arg->type->baseType == Type::tFloat4 ) {
                     arg->type->ref = false;
                 }
diff --git a/src/dasIMGUI_NODE_EDITOR.main.cpp b/src/dasIMGUI_NODE_EDITOR.main.cpp
index 54b5c34..57662ac 100644
--- a/src/dasIMGUI_NODE_EDITOR.main.cpp
+++ b/src/dasIMGUI_NODE_EDITOR.main.cpp
@@ -32,7 +32,7 @@ namespace das {
         for ( auto & pfn : this->functions.each() ) {
             bool anyString = false;
             for ( auto & arg : pfn->arguments ) {
-                if ( arg->type->constant && arg->type->ref && arg->type->dim.size()==0 ) {
+                if ( arg->type->constant && arg->type->ref && !arg->type->isArray() ) {
                     if ( arg->type->baseType==Type::tFloat2 || arg->type->baseType==Type::tFloat4 ) {
                         arg->type->ref = false;
                     }