code's small correction