A small correction to avoid a error generated in certain cases