Fix for *most* Theano test failures -- but not all :-(