diff --git a/nixos-infect b/nixos-infect
index c288b03..d38f1da 100755
--- a/nixos-infect
+++ b/nixos-infect
@@ -31,7 +31,7 @@ makeConf() {
 
   [[ -n "$doNetConf" ]] && network_import="./networking.nix # generated at runtime by nixos-infect"
   cat > /etc/nixos/configuration.nix << EOF
-{ ... }: 
+{ pkgs, lib, ... }: 
 {
   imports = [
     ./hardware-configuration.nix