Adds missing return