Fix for #41 (forgot to update part of the JS file.. doh!)