Added line to fix bug with return on main block.