Small bugfix in protocol