diff --git a/tests/unit/completion/test_models.py b/tests/unit/completion/test_models.py
index 32e6920fe..c3593031c 100644
--- a/tests/unit/completion/test_models.py
+++ b/tests/unit/completion/test_models.py
@@ -99,7 +99,7 @@ def configdata_stub(config_stub, monkeypatch, configdata_init):
             typ=configtypes.Dict(
                 keytype=configtypes.String(),
                 valtype=configtypes.Dict(
-                    keytype=configtypes.String(),
+                    keytype=configtypes.Key(),
                     valtype=configtypes.Command(),
                 ),
             ),
@@ -117,7 +117,7 @@ def configdata_stub(config_stub, monkeypatch, configdata_init):
             typ=configtypes.Dict(
                 keytype=configtypes.String(),
                 valtype=configtypes.Dict(
-                    keytype=configtypes.String(),
+                    keytype=configtypes.Key(),
                     valtype=configtypes.Command(),
                 ),
             ),