fix bug
version 2.0