Update test setup to call new integration function

Corrects test setup issue.