update to fix tests