Fix a bug with make_theano_batch call