[US031][Chethan, Pranav] Fixing a minor bug