Previous pull request missing commit

.. only a couple of line so made the change again.