Fix - restore function