Fix depth count in apply colocation