updated unit tests to actually work