Minor cleanup

Just