Fix missed case. Refactored a bit.
Merge branch 'force_protocol' of https://github.com/dawik/rebar into force_protocol
Oops