Minor assertion fix