add back code that got lost in translation