Add removed import