Fix to assert warning