Remove unnecessary part in models