Fix test as result varies depending on environment