Fix tests failing after scalatest upgrade

oops!