Fixed model import code.