Some minor cleanup

meh. Fixed now