test_utils: fix the axis list/tuple here as well