fix a checking bug

uiua-lang · Nov 27, 2024 · c51ae0f · c51ae0f
1 parent 7d5650f
commit c51ae0f
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/src/check.rs b/src/check.rs
@@ -521,16 +521,20 @@ impl VirtualEnv {
                     self.handle_args_outputs(f.args, f.outputs + 1);
                 }
                 prim if prim.modifier_args().is_some() => {
-                    return Err(SigCheckError::from(format!(
-                        "{} was not checked. This is a bug in the interpreter",
-                        prim.format()
-                    )))
+                    if let Some(sig) = prim.sig() {
+                        self.handle_sig(sig);
+                    } else {
+                        return Err(SigCheckError::from(format!(
+                            "{} was not checked. This is a bug in the interpreter",
+                            prim.format()
+                        )));
+                    }
                 }
                 prim => {
                     return Err(SigCheckError::from(format!(
                         "{} was checked as a modifier. This is a bug in the interpreter",
                         prim.format()
-                    )))
+                    )));
                 }
             },
             Node::ImplMod(prim, args, _) => match prim {