diff --git a/packages/builder/src/helpers/components.js b/packages/builder/src/helpers/components.ts
similarity index 98%
rename from packages/builder/src/helpers/components.js
rename to packages/builder/src/helpers/components.ts
index b1116d7750..f03551076e 100644
--- a/packages/builder/src/helpers/components.js
+++ b/packages/builder/src/helpers/components.ts
@@ -8,13 +8,14 @@ import {
 } from "@budibase/string-templates"
 import { capitalise } from "@/helpers"
 import { Constants } from "@budibase/frontend-core"
+import { ScreenProps } from "@budibase/types"
 
 const { ContextScopes } = Constants
 
 /**
  * Recursively searches for a specific component ID
  */
-export const findComponent = (rootComponent, id) => {
+export const findComponent = (rootComponent: ScreenProps, id: string) => {
   return searchComponentTree(rootComponent, comp => comp._id === id)
 }