Fix focus() on expressions

Test Plan:
 * subperseus
 * answer http://localhost:8080/preview/content/items/x3967fd785a804adb?version=edit incorrectly
 * make sure the check answer buttons shakes

Auditors: alex