Add annotation to fix math tests.

@ntoronto. My guess is that you were just testing the implementation of your procedures and not the inference of TR here and so this fix should be ok.
I'm still trying to work out if this a regression in TR or if it is a soundness fix and currently I believe it is an soundness fix.