[FIX] correct call to code, due to the rename of columns