Fixing bug with the migration to lodash