Fix transpose bug