Use the new helper which saves a temporary variable and a few lines of
code.
Signed-off-by: NMatthew Wilcox <mawilcox@microsoft.com>