Nice solution with `Assume` :+1: