Removed import of deleted file.
Updated test method.