Remove unused import
Relaxed test to pass with version 2.54