Fix compilation issue in test