Minor fix to support new LiteListener signature in test