small correction to code - partial model