diff --git a/packages/builder/src/builderStore/buildCodeForScreens.js b/packages/builder/src/builderStore/buildCodeForScreens.js
index 66512b703e..af8a930c93 100644
--- a/packages/builder/src/builderStore/buildCodeForScreens.js
+++ b/packages/builder/src/builderStore/buildCodeForScreens.js
@@ -27,7 +27,7 @@ export const buildCodeForScreens = screens => {
 }
 
 const buildComponentCode = componentProps =>
-  `"${componentProps._id}" : (render, context) => {
+  `"${componentProps._id}" : (render, context, state, routeParams) => {
 ${componentProps._code}
 },
 `