fixed up some unit tests
#44
#44