Fixed unit tests to stop the CI server complaining.