This logic would have to be updated after my comment above