Correction as per PR comment.