Small change in tests
Upgrade for build