Minor fix: Removed unneeded import