Update test
More changes to test