Add missing import of simtk.unit