Try to get unit tests to function again