one last fix for tests.