Fixed widget syntax highlighting

Made it compatible with recent syntax highlighting improvements.