small change to adapt to Bora's code