[AML] Fix pts error correction logic