Fix breaking API change in Theano 0.6