Fix center of mass motion removal with frozen atoms