Fixed bug in code section