diff --git a/bin/dtrace_profile.sh b/bin/dtrace_profile.sh
index 94f01e0..e197776 100755
--- a/bin/dtrace_profile.sh
+++ b/bin/dtrace_profile.sh
@@ -2,10 +2,10 @@
 # AUTHOR:  Jeremy Wall (jw), jeremy@marzhillstudios.com
 set -x
 
-file=$1
+cmd=$1
 
 cargo build --release
-sudo dtrace -c "target/release/ucg build ${file}" -o out.stacks -n 'profile-997 /execname == "ucg"/ { @[ustack(100)] = count(); }'
+sudo dtrace -c "target/release/ucg ${cmd}" -o out.stacks -n 'profile-997 /execname == "ucg"/ { @[ustack(100)] = count(); }'
 stackcollapse.pl out.stacks > collapsed.stacks
 cat collapsed.stacks | flamegraph.pl --minwidth 2.5 > perf_graph.svg
-rm -f out.stacks collapsed.stacks
\ No newline at end of file
+rm -f out.stacks collapsed.stacks