Added fix for junits I broke last commit.