Where should this logic go exactly? I don't really see a good place for it.