Fixing set shape rotation bug