diff --git a/lib/meat.ml b/lib/meat.ml
index 759e413..1e6c13d 100644
--- a/lib/meat.ml
+++ b/lib/meat.ml
@@ -62,8 +62,19 @@ let look () =
   do_cmd "nix flake show $NH_FLAKE";
   print_string footer
 
-let all_flag () = Array.mem "-a" argv || Array.mem "--all" argv
-let sub_flag () = Array.mem "-s" argv || Array.mem "--subflake" argv
+(* let all_flag () = Array.mem "-a" argv || Array.mem "--all" argv
+let sub_flag () = Array.mem "-s" argv || Array.mem "--subflake" argv *)
+let all_flag () = if Array.length argv >= 3 then match Array.get argv 2 with
+  | "-a" | "--all" -> true
+  | _ -> false
+else
+  false
+let sub_flag () = if Array.length argv >= 3 then match Array.get argv 2 with
+  | "-s" | "--sub" -> true
+  | _ -> false
+else
+  false
+
 let derelativise base = List.map (fun a -> base ^ "/" ^ a)
 
 let filter_dirs fullpath dirs =