Fix scalastyle checks