Fix dangling return statement