Improve code node type