Updated unit tests to account for protocol removal