Changed imports on test classes

changed imports on the test classes to match the new maven dependencies