Minor code clean-up

Some minor code clean-up