diff --git a/nixos/modules/security/google_oslogin.nix b/nixos/modules/security/google_oslogin.nix
index 227e3b5bc4b9..42dd3b22a005 100644
--- a/nixos/modules/security/google_oslogin.nix
+++ b/nixos/modules/security/google_oslogin.nix
@@ -1,7 +1,4 @@
 { config, lib, pkgs, ... }:
-
-with lib;
-
 let
 
   cfg = config.security.googleOsLogin;
@@ -13,8 +10,8 @@ in
 
   options = {
 
-    security.googleOsLogin.enable = mkOption {
-      type = types.bool;
+    security.googleOsLogin.enable = lib.mkOption {
+      type = lib.types.bool;
       default = false;
       description = ''
         Whether to enable Google OS Login.
@@ -32,7 +29,7 @@ in
 
   };
 
-  config = mkIf cfg.enable {
+  config = lib.mkIf cfg.enable {
     security.pam.services.sshd = {
       makeHomeDir = true;
       googleOsLoginAccountVerification = true;