Fixed the return type