fix for #334

fix for #334