Minor change to main.scala