Updated WakeGenerator to work with latest Triton SDK