another quick fix for test failure