For yield issue in IntelliJ Idea