Restore function removed by mistake