Restoring system partition in TWRP
You need to restore the /system, boot.img, stock recovery also.  Your data should be fine, but as always, you should back up just in case.  The OTA will check for all three of those things and if they are modified at all (like when rooting), the OTA will fail to flash on your device.