Import Correct Names in Tests

The tests were previously failing to compile due to name mismatches; these now compile and pass.