Having an issue getting IntelliJ to highlight ScalaMock 3.x + ScalaTest properly

ProxyMockIntelliJHighlightingTests.scala

package com.example

import org.scalatest.FunSuite
import org.scalamock.scalatest.proxy.MockFactory
import org.scalamock.scalatest.{MockFactory => RegularMockFactory}
import org.scalamock.proxy.ProxyMockFactory

trait Highlight

class ProxyMockIntelliJHighlightingBrokenTest extends FunSuite with MockFactory {
  test("unable to resolve mock symbol") {
    val foo = mock[Highlight]
  }
  test("super[ProxyMockFactory] won't compile here") {
    // val foo = super[ProxyMockFactory].mock[Highlight]
  }
}

class ProxyMockIntelliJHighlightingWorkaroundTest extends FunSuite with RegularMockFactory with ProxyMockFactory {
  test("super[ProxyMockFactory] resolves symbol") {
    val foo = super[ProxyMockFactory].mock[Highlight]
  }
}