Small fix to object restore functionality.