Set up proper testing environment