Forgot to run integration tests. Ran and made updates to accommodate.