Fix checkstyle

- Basically a workaround for IntelliJ's automatic import optimization