Further corrections to align_test script