Add lines missed in last commit