changed imports on test java file