BalajiN | Bug fix