Missed this code somehow from earlier commit