Scala style fix