Update model tests