Fixed broken test scenarios.
Fixed broken test scenarios.