Small cleanup of EM code