Fixing assumes in tests to work properly while running tests in remote mode