Fixed random forrest bug