Fixed use of wrong Scala class.