refactored 05 - expectation set up in production code