Update tests to troubleshoot integration