Fixed a scala-style error by removing unused imports