Fixed small js bug