small change in tests