Fixing issue on previous commit