Fixing test to disconnect properly