"are you guys ok with me first patch this since all users argmax errors are now in test" - not sure what you mean