Fix unexpected return compile error

Fix fatal compilation error