fix passing of difficulty into model