From old code, but fixed that missing line.