1 files changed, 5 insertions, 0 deletions
diff --git a/src/main/java/design/model/StrokeLeague.java b/src/main/java/design/model/StrokeLeague.java
index 487a2b5..d051fd0 100644
--- a/src/main/java/design/model/StrokeLeague.java
+++ b/src/main/java/design/model/StrokeLeague.java
@@ -1,6 +1,7 @@
 package design.model;
 
 import com.fasterxml.jackson.annotation.JsonCreator;
+import com.fasterxml.jackson.annotation.JsonTypeName;
 
 import java.util.ArrayList;
 import java.util.Collections;
@@ -10,6 +11,7 @@ import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
 
+@JsonTypeName("stroke")
 public class StrokeLeague extends League {
     private List<Golfer> participants;
     private Map<Golfer, Integer> totalStrokes;
@@ -44,6 +46,9 @@ public class StrokeLeague extends League {
     }
 
     @Override
+    public String getType() {
+        return "stroke";
+    }
     public void recordPlay(Golfer player, Match match, Round round){
         if(!isPlayable() || !participants.contains(player)) return;
         int strokes = round.getTotalSwings();