Scalastyle problem fixed