Fix missing return value