fix tests to fix compile error for broken build