1 files changed, 9 insertions, 4 deletions
diff --git a/test/src/model/PackageTest.kt b/test/src/model/PackageTest.kt
index 1f1a6ebf..701efbf0 100644
--- a/test/src/model/PackageTest.kt
+++ b/test/src/model/PackageTest.kt
@@ -1,8 +1,11 @@
 package org.jetbrains.dokka.tests
 
+import org.jetbrains.dokka.Content
+import org.jetbrains.dokka.DocumentationNode
+import org.jetbrains.kotlin.config.KotlinSourceRoot
 import org.junit.Test
-import kotlin.test.*
-import org.jetbrains.dokka.*
+import kotlin.test.assertEquals
+import kotlin.test.assertTrue
 
 public class PackageTest {
     Test fun rootPackage() {
@@ -45,7 +48,8 @@ public class PackageTest {
     }
 
     Test fun multipleFiles() {
-        verifyModel("test/data/packages/dottedNamePackage.kt", "test/data/packages/simpleNamePackage.kt") { model ->
+        verifyModel(KotlinSourceRoot("test/data/packages/dottedNamePackage.kt"),
+                    KotlinSourceRoot("test/data/packages/simpleNamePackage.kt")) { model ->
             assertEquals(2, model.members.count())
             with(model.members.single { it.name == "simple" }) {
                 assertEquals(DocumentationNode.Kind.Package, kind)
@@ -66,7 +70,8 @@ public class PackageTest {
     }
 
     Test fun multipleFilesSamePackage() {
-        verifyModel("test/data/packages/simpleNamePackage.kt", "test/data/packages/simpleNamePackage2.kt") { model ->
+        verifyModel(KotlinSourceRoot("test/data/packages/simpleNamePackage.kt"),
+                    KotlinSourceRoot("test/data/packages/simpleNamePackage2.kt")) { model ->
             assertEquals(1, model.members.count())
             with(model.members.elementAt(0)) {
                 assertEquals(DocumentationNode.Kind.Package, kind)