Fix missed return statement