Fixed my previous change to disconnecting (oops D:)