Fix highlighting.

Fixes #66