You are right, you need two hots going into the STC, I made that diagram in like the last 3 mins of my break... also where I wrote hot out it should be the cool circuit and not the heat circuit.
Your diagram looks correct, you have both Hot and Neutral going to the old circuit so you will...