Fix for #64

This brings back the yellow highlight.
Merge pull request #69 from wormeyman/patch-2

Fix for #64