hopefully fix tests