Fix pyrax exception import