Allowed wakeup to take an IStateBuilder