opps - fix tests