Remove the import that you don't need in both tests file.