added done call in after hook