S2W fix during boot

Signed-off-by: Umair Khan <omerjerk@gmail.com>