Fixed model import