[Pranav] A fix for the previous commit