Fixed the type of the return value