add back in some forgotten code