Small change to clean up imports