Test assertion fix