Fix another mock import
1 file changed