Fix up tw_reboot function

Change-Id: I53d329e493f4bcb3fe9ed177e7b2dede30ccf490