Changed tests to eliminate should_receive