Fix bug return