Rewrote test for ScalaTest