Fixed hack to the keys.

Accidentally removed a line I meant to edit, oops.