Correction needed for YAK

This fixes the issue in YAK, thanks @YaqiWang 
(closes #4389)