Fixed a minor bug
amits changes