Put loadbalancer mock somewhere more reusable