diff --git a/internal/util-complete/src/test/scala/sbt/complete/FileExamplesTest.scala b/internal/util-complete/src/test/scala/sbt/complete/FileExamplesTest.scala
index 173b276fc..f7b79573c 100644
--- a/internal/util-complete/src/test/scala/sbt/complete/FileExamplesTest.scala
+++ b/internal/util-complete/src/test/scala/sbt/complete/FileExamplesTest.scala
@@ -9,6 +9,7 @@ package sbt.internal.util
 package complete
 
 import java.io.File
+import org.scalatest.Assertion
 import sbt.io.IO
 
 class FileExamplesTest extends UnitSpec {
@@ -57,7 +58,8 @@ class FileExamplesTest extends UnitSpec {
   }
 
   def withDirectoryStructure[A](withCompletionPrefix: String = "")(
-      thunk: DirectoryStructure => A): Unit = {
+      thunk: DirectoryStructure => Assertion
+  ): Assertion = {
     IO.withTemporaryDirectory { tempDir =>
       val ds = new DirectoryStructure(withCompletionPrefix)
       ds.createSampleDirStructure(tempDir)