Fixed missed code section