Trying to get unit tests to work