Fix missing return type