diff --git a/stylix/palette.nix b/stylix/palette.nix
index af6cfe1..a3c8874 100644
--- a/stylix/palette.nix
+++ b/stylix/palette.nix
@@ -44,14 +44,14 @@ in {
 
       primaryScale = {
         dark = lib.mkOption {
-          type = lib.types.number;
+          type = lib.types.float;
           default = 1;
           description = ''
             Use this option to scale color values for the generated light theme.
           '';
         };
         light = lib.mkOption {
-          type = lib.types.number;
+          type = lib.types.float;
           default = 1;
           description = ''
             Use this option to scale color values for the generated light theme.