Agent is leaking connections to storage

2019-07-23 08:23:24,664 ERROR:storage.py:675(2746): Remote storage operation failed (request: 'None')
ESC[31mTraceback (most recent call last):
  File "src/storage.py", line 582, in receiveMessageFromFile
    data = fileObj.read(struct.calcsize("!I"))
  File "/usr/lib/python3.6/socket.py", line 586, in readinto
    return self._sock.recv_into(b)
ConnectionResetError: [Errno 104] Connection reset by peerDuring handling of the above exception, another exception occurred:Traceback (most recent call last):
  File "src/storage.py", line 656, in handle
    message = _ArtifactSocketHandler.receiveMessageFromFile(self.rfile, messages.FileTransferRequest)
  File "src/storage.py", line 589, in receiveMessageFromFile
    log.info("Connection reset while receiving file from storage '%s'.", message.storageLocation)
UnboundLocalError: local variable 'message' referenced before assignmentESC[0m