Fix for #64

This brings back the yellow highlight.