Made predict work with HAWQ