Added doc
Fix test to pass with new refact