Fix mock import.