update testing code for LHE