Mock correctly