Add hack to fix test.