Changed junit  import