Fixes #18 issue with Brute Force argmaxer