Re-add unused import