Highlight doesn't disappear when going back

[sfsam]

Steps to reproduce:

1. Make a move.
2. Click on a piece to make another move. Observe that valid squares are highlighted in yellow.
3. Click the back button to undo the move from Step 1.
4. Observe that the squares from Step 2 are still highlighted.