Updated API mock