#36 updated unit test