Small fix in test scenario