Fixed bug with exercise
Signed-off-by:Andrei Dincescu <andrei.dincescu@hotmail.com>