Minor cleanup fix