Update for CI tests