Fixed broken test setup files