diff --git a/python-api-src/library.cpp b/python-api-src/library.cpp
index 9b231c4a3ecb292458a8ae9bd5f53a6713a75e3b..e183d8d498b49251a4c20176fb932651459487f2 100644
--- a/python-api-src/library.cpp
+++ b/python-api-src/library.cpp
@@ -127,8 +127,16 @@ PYBIND11_MODULE(library, m)
         .value("CheatMoney", sc2::Difficulty::CheatMoney)
         .value("CheatInsane", sc2::Difficulty::CheatInsane);
 
-	m.def("create_participants", &sc2::CreateParticipant, "Create participant from bot", "race"_a, "bot"_a, "player_name"_a);
-	m.def("create_computer", &sc2::CreateComputer, "Create participant from built-in Starcraft computer", "race"_a, "difficulty"_a, "ai_build"_a, "player_name"_a);
+	py::enum_<sc2::AIBuild>(m, "AIBuild")
+		.value("RandomBuild", sc2::AIBuild::RandomBuild)
+		.value("Rush", sc2::AIBuild::Rush)
+		.value("Timing", sc2::AIBuild::Timing)
+		.value("Power", sc2::AIBuild::Power)
+		.value("Macro", sc2::AIBuild::Air);
+
+
+	m.def("create_participants", &sc2::CreateParticipant, "Create participant from bot", "race"_a, "bot"_a, "player_name"_a = "");
+	m.def("create_computer", &sc2::CreateComputer, "Create participant from built-in Starcraft computer", "race"_a, "difficulty"_a, "ai_build"_a = sc2::AIBuild::RandomBuild, "player_name"_a = "");
 
     /*
     py::class_<BuildDescription>(m, "BuildDescription")