Cleaned up how I saved the sock reference

This just made more sense