fixed tests with a recent change