Change junit to a test dependency