Fix build (Thank you IntelliJ for adding stupid import)