fix bug from 3.0.1