Minor modification in the helper file