Fixing another bug from the connection refactoring.