Try to fix small js bug