Revert "Remove model dependency as it no longer exists"

This reverts commit a5b75d49996c18ea60eda83573e9f55318dca6c3.