diff --git a/src/function.py b/src/function.py
index c9e44a3..fb17b85 100644
--- a/src/function.py
+++ b/src/function.py
@@ -5,10 +5,10 @@ Use the automation_context module to wrap your function in an Automate context h
 
 from speckle_automate import AutomationContext, AutomateBase
 
-from rules import apply_rules_to_objects
-from inputs import FunctionInputs
-from helpers import flatten_base
-from spreadsheet import read_rules_from_spreadsheet
+from src.rules import apply_rules_to_objects
+from src.inputs import FunctionInputs
+from src.helpers import flatten_base
+from src.spreadsheet import read_rules_from_spreadsheet
 
 
 def automate_function(
diff --git a/src/rules.py b/src/rules.py
index 7525539..1ae98b8 100644
--- a/src/rules.py
+++ b/src/rules.py
@@ -5,7 +5,7 @@ from Levenshtein import ratio
 import pandas as pd
 import re
 
-from helpers import speckle_print
+from src.helpers import speckle_print
 
 
 # We're going to define a set of rules that will allow us to filter and