Fix predictions not being processed