Fixed a disconnect issue