Fixed a bug.

Signed-off-by: Harman Sohanpal <harmansohanpal@gmail.com>