Fixing "shouldReceive" method definition