are you guys ok with me first patch this since all users argmax errors are now in test