fix axis transposition error