fixed bug in bart predict function