Fixed missing return

It seems I have a serious problem with returns...