Revert "Revert "fix test""

This reverts commit b3eb6650d5b4a3849ff35124ee149d5610654fef.