small edit in models