Fixed return statement

Needed to propagate it further