Fixed test task

Just for passing it...