Fix assert code