Fixing missing return type