Small Mistake

Updated disconnect code