diff --git a/cookbooks/aws-parallelcluster-shared/resources/activate_virtual_env.rb b/cookbooks/aws-parallelcluster-shared/resources/activate_virtual_env.rb
index 6ab1ce6e06..05818c2847 100644
--- a/cookbooks/aws-parallelcluster-shared/resources/activate_virtual_env.rb
+++ b/cookbooks/aws-parallelcluster-shared/resources/activate_virtual_env.rb
@@ -41,3 +41,7 @@
     end
   end
 end
+
+action_class do
+  include PyEnv::Cookbook::ScriptHelpers
+end