Fixed one of the specs failing