Update the integration test to not rely on the presentation class name