diff --git a/src/cb_dasIMPLOT.h b/src/cb_dasIMPLOT.h
index 1bc40ec..910e4b5 100644
--- a/src/cb_dasIMPLOT.h
+++ b/src/cb_dasIMPLOT.h
@@ -32,7 +32,7 @@ struct implotTempFn {
         }
         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/dasIMPLOT.main.cpp b/src/dasIMPLOT.main.cpp
index 875cfea..7570d63 100644
--- a/src/dasIMPLOT.main.cpp
+++ b/src/dasIMPLOT.main.cpp
@@ -314,7 +314,7 @@ void Module_dasIMPLOT::initMain () {
     // mirrors dasImguiNodeEditor's initMain fixup.
     for ( auto & pfn : this->functions.each() ) {
         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;
                 }