Fixed issue: Fixed mistake made in the previous commit