Fix wake up