Fix rotation bug