I have a problem I have an ability that takes control of a unit, that works but after the wait time I put in it doesn't change back this is what I used.
Unit - Change ownership of (Target unit of ability being cast) to (Previous owner) and Change color
Unit - Change ownership of (Target unit of ability being cast) to (Previous owner) and Change color