Fix reverted issue