Restore accidentally removed use statements