Fixed IntelliJ warnings