Change afterTest extension type to Closure