Changed stage client to allow disconnect