junit.framework -> org.junit & fixed a broken test