[Debugging] Fix return type of test function