Revert "Revert "Added one line of code""

This reverts commit c19490448c8955a5d9048d64bde860f14791c511.