Fixed a bug in the js