Fix notebook highlighting