Attempting to get the tests to work