Set up tests properly