How do I straighten out character angle after jump?

-Player is a square

-I have platform behavior and a solid object floor.

-Spacebar simulates jump

-When player not on floor I rotate

When I press spacebar the square jumps and rotates how I want but when it lands back on the floor or onto another square it doesn't flatten out. What am I missing here? Thanks for any advice!