Fix js bug :(