Modification to correctly use @Async.