improved clean after

Signed-off-by: Igor Gnatenko <i.gnatenko.brain@gmail.com>