diff --git a/gen_makefile.sh b/gen_makefile.sh
index f5150fc..8960183 100755
--- a/gen_makefile.sh
+++ b/gen_makefile.sh
@@ -1,13 +1,13 @@
 #!/bin/bash
 
 files=()
-dir=$1
-args=("$@")
+dir=$1     #Base directory
+types=($2) #List of types (convert to array)
 
 echo "openpower_procedures_cpp_files = \\"
-for ((i=1; i<${#args[@]}; ++i));
+for ((i=0; i<${#types[@]}; ++i));
 do
-    type=${args[$i]}
+    type=${types[$i]}
     type=${type// /} #remove spaces
     for file in $(ls $dir/procedures/$type/*.cpp);
     do