Fix attribute name in after task