Fix rotation code