Fixed 2.0.4 error