Removed mock import

removed to make work on RTD