Fixed test for unfinished piece of code