Fixing missing return value