Fix build problem in IntelliJ