Fix reverted code.