Missing code to fix. closes #2755