Added small change to test