Missed out change from last commit