Fix IntelliJ autocomplete issue