Fixed a missing include for test post-processing