missed an important line, fixed