diff --git a/main.cpp b/main.cpp
index 98630b0..e5e6f69 100644
--- a/main.cpp
+++ b/main.cpp
@@ -14,7 +14,7 @@
 
 
 namespace Hanabi {
-    void download(int game_id, int turn) {
+    void download(std::variant<int, const char*> game_id, int turn) {
         auto game = Download::get_game(game_id, turn);
         std::cout << "Analysing state: " << std::endl << *game << std::endl;
         auto res = game->backtrack(1);
@@ -109,11 +109,15 @@ int main(int argc, char *argv[]) {
     #ifndef NDEBUG
       test();
     #endif
-    check_games(4, 9);
+    check_games(2, 9);
   if(argc == 3) {
-      std::string game (argv[1]);
+      std::string game(argv[1]);
       std::string turn (argv[2]);
-      Hanabi::download(std::stoi(game), std::stoi(turn));
+      try {
+          Hanabi::download(std::stoi(game), std::stoi(turn));
+      } catch(std::invalid_argument&) {
+          Hanabi::download(game.c_str(), std::stoi(turn));
+      }
   } else {
       Hanabi::print_usage(argv[0]);
   }