Fix compilation issue in test module