fix tests a bit I guess