Adds missing return statement