Minor alteration to a test