Fixed  broken test