Update test implementations file to match production version