Adde missing return statement