Fix assert error