Fix warnings in IntelliJ