1 files changed, 4 insertions, 2 deletions
diff --git a/bundle-plugin/src/test/scala/com/yahoo/container/plugin/classanalysis/AnalyzeMethodBodyTest.scala b/bundle-plugin/src/test/scala/com/yahoo/container/plugin/classanalysis/AnalyzeMethodBodyTest.scala
index 231aba36325..43f52884f39 100644
--- a/bundle-plugin/src/test/scala/com/yahoo/container/plugin/classanalysis/AnalyzeMethodBodyTest.scala
+++ b/bundle-plugin/src/test/scala/com/yahoo/container/plugin/classanalysis/AnalyzeMethodBodyTest.scala
@@ -1,17 +1,19 @@
 // Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
 package com.yahoo.container.plugin.classanalysis
 
-import org.scalatest.junit.{JUnitSuite, ShouldMatchersForJUnit}
+import org.scalatest.junit.{AssertionsForJUnit, JUnitSuite}
 import sampleclasses._
 import TestUtilities._
 import org.junit.Test
 import java.io.PrintStream
 
+import org.scalatest.Matchers
+
 /**
  * Tests that classes used in method bodies are included in the imports list.
  * @author  tonytv
  */
-class AnalyzeMethodBodyTest extends JUnitSuite with ShouldMatchersForJUnit {
+class AnalyzeMethodBodyTest extends JUnitSuite with AssertionsForJUnit with Matchers {
   @Test def require_that_class_of_locals_are_included() {
     analyzeClass[Methods].referencedClasses should contain(name[Base])
   }