fixed first unit test... more