Fix last test (should be green now)