WithGiven method corrections