Fixing  test build break