fix a major bug