fix 4.0.4 bug