Fix bug: run forever