Change the function def for comparison purpose