Fix to actually select the correct equation

Test plan: as before - check a few example exercises

Auditor: alpert