diff --git a/integration_test/run_tests.sh b/integration_test/run_tests.sh
index b90cb4ebb..e9ccbac00 100755
--- a/integration_test/run_tests.sh
+++ b/integration_test/run_tests.sh
@@ -104,7 +104,8 @@ function cleanup {
   rm $DIR/functions/firebase-functions.tgz
   rm $DIR/functions/package.json
   rm -f $DIR/functions/firebase-debug.log
-  rm -rf $DIR/functions/node_modules/firebase-functions
+  rm -rf $DIR/functions/lib
+  rm -rf $DIR/functions/node_modules
 }
 
 # Setup