Fix layer highlighting