diff --git a/lib/cli/bin/index.js b/lib/cli/bin/index.js
index df4a5a1221..136e6d73d6 100755
--- a/lib/cli/bin/index.js
+++ b/lib/cli/bin/index.js
@@ -1,3 +1,3 @@
-#!/usr/bin/env node -r esm
+#!/usr/bin/env node
 
-import './generate';
+require('esm')(module)('./generate.js');
diff --git a/lib/cli/test/run_tests.sh b/lib/cli/test/run_tests.sh
index 34908300c6..8f7c0c7ad2 100755
--- a/lib/cli/test/run_tests.sh
+++ b/lib/cli/test/run_tests.sh
@@ -36,10 +36,10 @@ do
   if [ $dir == *"native"* ]
   then
     # run @storybook/cli
-    ../../../bin/index.js init --skip-install --yes --install-server
+    yarn sb init --skip-install --yes --install-server
   else
     # run @storybook/cli
-    ../../../bin/index.js init --skip-install --yes
+    yarn sb init --skip-install --yes
   fi
 
   cd ..