Fix rotation behavior