Remove unused import and slightly update test