Fix scala style!