Minor mock api fix.