Refactor ShouldReceive
Correct missing/unnecessary requires