Remove excess import