diff --git a/neomi.py b/neomi.py index 4d29906..cdd67ef 100644 --- a/neomi.py +++ b/neomi.py @@ -105,7 +105,8 @@ def SocketReader(sock): if command is not None: if command == SocketReaderCommands.stop: - return + # Return the rest of data in buffer + return chunk else: raise CommandError('%s not recognised' % repr(command))