updated unit test to be correct