fixed junit problems