Fix for Square rotation

fixed the vroken square rotation which was caused by the last commit