Fix version numbering key replacement
Remove unneeded imports