diff --git a/python-api-src/library.cpp b/python-api-src/library.cpp
index 7f54cb213c2bc2c2a714b3ad8631441f8c54fa84..020208d32fe76584a587c4b48f96cde4639819d9 100644
--- a/python-api-src/library.cpp
+++ b/python-api-src/library.cpp
@@ -38,7 +38,6 @@ PYBIND11_MODULE(library, m)
     TODO: These also needs to be defined
 
     typedef SC2Type<ABILITY_ID>   AbilityID;
-    typedef SC2Type<UPGRADE_ID>   UpgradeID;
     typedef SC2Type<BUFF_ID>      BuffID;
     */
     py::class_<sc2::UnitTypeID>(m, "UnitTypeID")
@@ -46,6 +45,11 @@ PYBIND11_MODULE(library, m)
 
     py::implicitly_convertible<sc2::UNIT_TYPEID, sc2::UnitTypeID>();
 
+    py::class_<sc2::UpgradeID>(m, "UpgradeID")
+        .def(py::init<sc2::UPGRADE_ID>());
+
+    py::implicitly_convertible<sc2::UPGRADE_ID, sc2::UpgradeID>();
+
     py::class_<sc2::Agent>(m, "Agent")
         .def(py::init());