We'll need to rework this logic later.