Fix test compilation issue