Fix #8, the socket module haven't been imported yet.