Change the connection object too

Forgot this... without it the class changes, but the connection doesn't actually change from the mock connection.