Switching prax to unstable branch implementation