Since the `get_write_location` is used only in `LB` context this moves it to be only available there.
