diff --git a/python-api-src/lib_unit.cpp b/python-api-src/lib_unit.cpp index 3ac60ab5150c966cd0d34dcc167f39cdc25fa631..fe68d9fbd75b3cf8bb352137e920dcd02361e720 100644 --- a/python-api-src/lib_unit.cpp +++ b/python-api-src/lib_unit.cpp @@ -37,7 +37,7 @@ void define_unit(py::module & m) .def("build_target", &Unit::buildTarget) .def("train", &Unit::train) .def("morph", &Unit::morph) - .def("__hash__", [](const Unit & unit) { return std::hash<CCUnitID>{}(unit.getID()); }) + .def("__hash__", [](const Unit & unit) { return std::hash<const sc2::Unit *>{}(unit.getUnitPtr()); }) .def(py::self == py::self) .def("__repr__", [](const Unit & unit) { return "<Unit of type: '" + unit.getType().getName() + "'>"; }); }