If you have any ideas how we can better unit test this, I'm all ears. PR welcome :)