Fix some warnings from IntelliJ