diff --git a/saqc/core/register.py b/saqc/core/register.py
index e0e6a443378eea4f0744d59b2429be55999079f8..6654d68d060b5e35f08a390562cd2361683887aa 100644
--- a/saqc/core/register.py
+++ b/saqc/core/register.py
@@ -56,6 +56,8 @@ def register(masking: MaskingStrT = "all", module: Optional[str] = None):
             return _postCall(result, old_state)
 
         FUNC_MAP[func_name] = SaQCFunction(func_name, callWrapper)
+        callWrapper._module = module
+        callWrapper._masking = masking
 
         return callWrapper