Bug fix for issue #681

Add missing line of code