1 files changed, 7 insertions, 1 deletions
diff --git a/src/main.cpp b/src/main.cpp
index c3945d3..98ff360 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -33,7 +33,12 @@ static void perft(int argc, char *argv[]) {
   auto start = std::chrono::steady_clock::now();
 
   Game game = initBoard(fen, nullptr);
-  uint64_t count = MoveGenTest(depth, &game);
+  uint64_t count = 0;
+#ifdef PERFT_DIVIDE
+  count = MoveGenTestDivide(depth, &game);
+#else
+  count = MoveGenTest(depth, &game);
+#endif
 
   auto end = std::chrono::steady_clock::now();
   auto ms = std::chrono::duration_cast<std::chrono::milliseconds>(end - start)
@@ -72,6 +77,7 @@ static void zobristCmd(int argc, char *argv[]) {
 
 int main(int argc, char *argv[]) {
   InitZobrist();
+  initMagicBitboards();
 
   if (argc >= 2 && std::string(argv[1]) == "perft") {
     perft(argc, argv);