fix model

undoing something i was experimenting with