diff --git a/tests/games.py b/tests/games.py
index 6fe0ed10a..652dafc73 100644
--- a/tests/games.py
+++ b/tests/games.py
@@ -5,7 +5,12 @@
 
 
 def read_from_file(fn: str) -> gbt.Game:
-    return gbt.Game.read_game(pathlib.Path("tests/test_games")/fn)
+    if fn.endswith(".efg"):
+        return gbt.read_efg(pathlib.Path("tests/test_games")/fn)
+    elif fn.endswith(".nfg"):
+        return gbt.read_nfg(pathlib.Path("tests/test_games")/fn)
+    else:
+        raise ValueError(f"Unknown file extension in {fn}")
 
 
 ################################################################################################