- Changed a function return type to fix a wrong type error