diff --git a/spacy/tests/regression/test_issue595.py b/spacy/tests/regression/test_issue595.py
index 6c73a621a..4a83a4020 100644
--- a/spacy/tests/regression/test_issue595.py
+++ b/spacy/tests/regression/test_issue595.py
@@ -12,7 +12,7 @@ import pytest
 def test_issue595():
     """Test lemmatization of base forms"""
     words = ["Do", "n't", "feed", "the", "dog"]
-    tag_map = {'VB': {POS: VERB, 'morph': VerbForm_inf}}
+    tag_map = {'VB': {POS: VERB, VerbForm_inf: True}}
     rules = {"verb": [["ed", "e"]]}
 
     lemmatizer = Lemmatizer({'verb': {}}, {'verb': {}}, rules)