Added another socket test + cosmetic fix