Fix Pylearn 2.6 compat problem