Fixed minor JS issue