Small import clean up

Nothing special, just a small cleanup