diff --git a/java/kotlin-extractor2/src/main/kotlin/entities/Function.kt b/java/kotlin-extractor2/src/main/kotlin/entities/Function.kt
index a919a8232fa..1d7e2c5a9e2 100644
--- a/java/kotlin-extractor2/src/main/kotlin/entities/Function.kt
+++ b/java/kotlin-extractor2/src/main/kotlin/entities/Function.kt
@@ -1,6 +1,5 @@
 package com.github.codeql
 
-import com.github.codeql.KotlinUsesExtractor.TypeContext
 import org.jetbrains.kotlin.analysis.api.KaSession
 import org.jetbrains.kotlin.analysis.api.symbols.KaFunctionSymbol
 import org.jetbrains.kotlin.analysis.api.symbols.name
@@ -723,4 +722,4 @@ OLD: KE1
             }
     */
     return id
-}
\ No newline at end of file
+}
diff --git a/java/kotlin-extractor2/src/main/kotlin/entities/Types.kt b/java/kotlin-extractor2/src/main/kotlin/entities/Types.kt
index 32845b156d0..7b297da0971 100644
--- a/java/kotlin-extractor2/src/main/kotlin/entities/Types.kt
+++ b/java/kotlin-extractor2/src/main/kotlin/entities/Types.kt
@@ -1,6 +1,5 @@
 package com.github.codeql
 
-import com.github.codeql.KotlinUsesExtractor.TypeContext
 import org.jetbrains.kotlin.analysis.api.types.KaClassType
 import org.jetbrains.kotlin.analysis.api.types.KaType
 
@@ -50,7 +49,7 @@ private fun extractErrorType(): TypeResults {
 */
 
 // TODO
-fun fakeKotlinType(): Label<out DbKt_type> {
+fun KotlinUsesExtractor.fakeKotlinType(): Label<out DbKt_type> {
     val fakeKotlinPackageId: Label<DbPackage> =
         tw.getLabelFor("@\"FakeKotlinPackage\"", { tw.writePackages(it, "fake.kotlin") })
     val fakeKotlinClassId: Label<DbClassorinterface> =