Removing import that's breaking the build