Swells fix broken test