diff --git a/cpp/src/main.cpp b/cpp/src/main.cpp
index 834e87d..3c18d61 100644
--- a/cpp/src/main.cpp
+++ b/cpp/src/main.cpp
@@ -1,11 +1,14 @@
-#include "stickfosh.hpp"
+#include "ais/ai.hpp"
 
 #include <iostream>
 
 int main(int argc, char* argv[]) {
     std::string pos =
         "rnbqkbnr/pppppppp/8/8/8/8/PPPPPPPP/RNBQKBNR w KQkq - 0 1";
-    std::string move = search(pos, 4);
+
+    ai::v1_simple ai;
+
+    std::string move = ai.search(pos, 4);
     std::cout << move << std::endl;
     return 0;
 }