Bug/junit imports

hi! just one small change. After merging pull request #75 the unit test wasn't compiling because of some missing junit imports.