Fixing integration tests I didn't check last time I committed