Fixed faulty mock import