trying too fix tests