Decided to use it in both branches