fix compile error on test