Small fix in test