fixed first test!! :)