fix tests broken in previous commit