Fixes function missing return value