Updated mock tests