Remove breaking import