Fixed IntelliJ code bug