tests

just fixed up some unit tests