diff --git a/hosts/Infini-SWIFT/default.nix b/hosts/Infini-SWIFT/default.nix
index 8ab2906..d7db555 100644
--- a/hosts/Infini-SWIFT/default.nix
+++ b/hosts/Infini-SWIFT/default.nix
@@ -1,5 +1,5 @@
-{ suites, profiles, pkgs, lib, ... }: {
-  imports = lib.flattenListSet {
+{ suites, profiles, pkgs, lib, test, ... }: {
+  imports = test.flattenListSet {
     suites = suites.graphic;
     imports = [ ./hardware-configuration.nix ];
     profiles = with profiles;
diff --git a/users/infinidoge/default.nix b/users/infinidoge/default.nix
index 21ee1b7..0d0fd50 100644
--- a/users/infinidoge/default.nix
+++ b/users/infinidoge/default.nix
@@ -1,6 +1,6 @@
-{ config, self, lib, pkgs, ... }: {
+{ config, self, lib, pkgs, test, ... }: {
   home-manager.users.infinidoge = { suites, profiles, ... }: {
-    imports = lib.our.flattenListSet {
+    imports = test.flattenListSet {
       suites = with suites; [ base ];
     };