Set up tests correcty