minor change to fix tests