Accidentally put something in a conditional I shouldn't have. Now works for both examples we have.