Fixed the Accidental delete

Fixed the Accidental delete of the latest code of @mtwebster