Change junit back to test dependency