Fix shapes of test files