def METHOD_NAME(self, entity):
    desired_z = self.map.get_z(entity.x, entity.y)
    if int(entity.z) != int(desired_z):
        entity.z = desired_z
        return True
    return protocol.METHOD_NAME(self, entity)