Fix missing returns statements.