diff --git a/modules/zyxel-dual-image/service.nix b/modules/zyxel-dual-image/service.nix
index b711d1a..3edd642 100644
--- a/modules/zyxel-dual-image/service.nix
+++ b/modules/zyxel-dual-image/service.nix
@@ -3,7 +3,7 @@
 , lib
 , zyxel-bootconfig
 }:
-{ ensureActiveImage, bootConfigurationMtdPartition, kernelCommandLineSource }:
+{ ensureActiveImage, primaryMtdPartition, secondaryMtdPartition, bootConfigurationMtdPartition, kernelCommandLineSource }:
 let
   inherit (liminix.services) oneshot;
   activeImageIndex = if ensureActiveImage == "primary" then 0 else 1;