revert change and add function
Fix