Fixed JS bug caused by the last commit. Oops