diff --git a/python-api-src/library.cpp b/python-api-src/library.cpp
index c868e9f561551180ee14bf0d5a34a7b7d8dea1f5..e8d5ff9975a194a93587c6384bd545b61eadc30f 100644
--- a/python-api-src/library.cpp
+++ b/python-api-src/library.cpp
@@ -79,9 +79,9 @@ PYBIND11_MODULE(library, m)
         .def_property_readonly("current_supply", &IDABot::GetCurrentSupply, "How much supply we are currently using")
         .def_property_readonly("max_supply", &IDABot::GetMaxSupply, "How much supply we can currently use")
         .def_property_readonly("gas", &IDABot::GetGas, "How much gas we currently have")
-        .def("use_ability", &IDABot::CallUnitCommand, "Use an ability with the given unit")
-        .def("use_ability", &IDABot::CallUnitCommandWithPoint, "Use ability at point with the given unit")
-        .def("use_ability", &IDABot::CallUnitCommandWithTarget, "Use ability at target with the given unit");
+        .def("use_ability", &IDABot::UseAbility, "Use an ability with the given unit")
+        .def("use_ability", &IDABot::UseAbilityWithPoint, "Use ability at point with the given unit")
+        .def("use_ability", &IDABot::UseAbilityWithTarget, "Use ability at target with the given unit");
 
     py::class_<sc2::PlayerSetup>(m, "PlayerSetup");