Fixed the socket stuff to work with the new UI also