[Unit Tests] Improve tests

* Model tests were creating duplicate objects