Fixing this to work for intellij