Fix safe_walk test