Minor cleanup 

Missed this