corrected import statement which was causing tests to fail