change import statement of mock module for rtd