Highlight Box fix