Fix the issue of the function never used (moved implementation to the test module)