Correct test imports. Thanks to DaveC for showing me the way.