Attempt to fix test failing on ci build