Fixed gain issue, missed the <vector> include