Fix to assert