Move another assert after check