diff --git a/src/Transformations/ReplaceArraysInIO/replace_dist_arrays_in_io.cpp b/src/Transformations/ReplaceArraysInIO/replace_dist_arrays_in_io.cpp
index 021e797..248cea2 100644
--- a/src/Transformations/ReplaceArraysInIO/replace_dist_arrays_in_io.cpp
+++ b/src/Transformations/ReplaceArraysInIO/replace_dist_arrays_in_io.cpp
@@ -19,7 +19,7 @@ using std::pair;
 
 static inline bool isArrayDeclaration(SgStatement* st)
 {
-    return isSgDeclarationStatement(st) || st->variant() == DIM_STAT || st->variant() == COMM_STAT;
+    return isSgDeclarationStatement(st) || isSgVarListDeclStmt(st) || isSgNestedVarListDeclStmt(st);
 }
 
 static SgExpression* findExprWithVariant(SgExpression* exp, int variant)
@@ -855,7 +855,7 @@ void replaceDistributedArraysInIO(vector<ParallelRegion*>& regions,
                         if (last_io_bound == curr_stmt) // always true
                             last_io_bound = curr_stmt->lexPrev();
                     }
-                    
+
                     curr_stmt = curr_stmt->lexNext();
                 }
             }