Fixed the errors caused by missing docstring