fix tests a bit.