Cleaned up how I saved the sock reference