That wiring diagram is correct. I plug my box into a wall, GCFI, plug my pump into the box and my heat element into the other side.
The screws are in a different position from some manufacturers of switch plug combos, but I colored them in the correct position from the receptacles that I found...