Hack: not training 0 anymore