*Fix reboot step - add assertion