Fix missed case. Refactored a bit. Changed from force_protocol to protocol