Fixed a bug in the last update