Reduce if condtion for finds

Remove the $state === 'after' condition.
Can only be before / after.