Fixing build break in test cases