Fixing imports from a silly intellij removal