[FIX] adapt align module tests to alignContext