update to fix failing tests