Fixed broken junit imports