diff --git a/01-data-model/data-model.ipynb b/01-data-model/data-model.ipynb
index f525899..56dd7eb 100644
--- a/01-data-model/data-model.ipynb
+++ b/01-data-model/data-model.ipynb
@@ -443,11 +443,10 @@
     }
    ],
    "source": [
-    "suit_values = dict(spades=3, hearts=2, diamonds=1, clubs=0)\n",
-    "\n",
     "def spades_high(card):\n",
     "    rank_value = FrenchDeck.ranks.index(card.rank)\n",
-    "    return rank_value * len(suit_values) + suit_values[card.suit]\n",
+    "    suit_value = FrenchDeck.suits.index(card.suit)\n",
+    "    return rank_value * len(FrenchDeck.suits) + suit_value\n",
     "\n",
     "for card in sorted(deck, key=spades_high):\n",
     "    print(card)"