Fixed error on the done()