Fix context when loading models