Corrected bug found by travis