Fixed bug with earlier commit

Signed-off-by: Pratheek Rebala <pratheekrebala@live.com>