Minor fix.

Must have merged some code wrong.