mahendra | fixed a minor bug