Fix scala style issue