Small code change to reduce lines no.