 
     @discardableResult
     public func expect() -> Expectation {
+
+        // flush all the argument configurations from the storage
+        ArgumentStorageImpl.instance.flush()
+
         let stub = Stub()
         let expectation = self.expectationFactory.expectation(withStub: stub)
 

    @discardableResult
    public func stub() -> Stub {
        let stub = Stub()

        // mark instance as being ready for registration
        self.stubBeingRegistered = stub

        // perform actual registration
        let ret: T? = self.register(function, with: argsConfig)


        return ret
    }
