Fixed code that was removed.