Fixed another assert