fix code highlight bug for now
fix code hightlight bug for now