Fix import into intellij