Add compatibility with pyrax 1.4.6