One changed missed out in the pull

Missed out in the previous pull due to some merge issues.