Fix unused import warning