Add a check for array for consume_after