fix: A lost connection bug