try and fix tests