Remove unused import in setup