Minor bugfix in protocol implementation