Former WWE star says he thought he was getting fired before training The Rock for his first return