Fix forgetting return call error