fix tests - not sure when this went bad...