Make small correction in code