Add fix a bug