Fix tests to be able to run.