The exit condition was missing.
Signed-off-by: Timothy Gu <timothygu99@gmail.com>
(cherry picked from commit e85d38c2)
Signed-off-by: Timothy Gu <timothygu99@gmail.com>