fix code class