Why did you remove this import? It makes the tests very ugly.