diff --git a/configuration.nix b/configuration.nix
index 324012b..2f35ebb 100644
--- a/configuration.nix
+++ b/configuration.nix
@@ -1,7 +1,8 @@
-{ config, pkgs, ... }:
+{ lib, config, pkgs, ... }:
 
 let
   hostname = "virtbox";
+  isVirtbox = hostname == "virtbox";
 in
 {
   imports =
@@ -33,7 +34,7 @@ in
 
   # branch and enable different capabilities based on the system
 
-  lib.mkIf hostname == "virtbox" {
+  test = lib.mkIf isVirtbox {
     services.virtbox.enable = true;
   };
 }
diff --git a/virtbox.nix b/virtbox.nix
index 05d75a9..ee649c5 100644
--- a/virtbox.nix
+++ b/virtbox.nix
@@ -1,4 +1,4 @@
-{ config, pkgs, ...}:
+{ lib, config, pkgs, ...}:
 
 with lib;
 let