updated for assembly ref fix