Fix timeout not properly working unless True is returned