updated test setup