diff --git a/modules/system/home/niri-setup/default.nix b/modules/system/home/niri-setup/default.nix
index 224568d..3e1f42d 100644
--- a/modules/system/home/niri-setup/default.nix
+++ b/modules/system/home/niri-setup/default.nix
@@ -1,4 +1,4 @@
-{ pkgs, ... }: {
+{ lib, config, pkgs, ... }: lib.mkIf (config.withNiri) {
   home.packages = with pkgs; [
     polkit_gnome
     dconf