Some fixes for GP regression