Fixing a few JS mismatches

@wedaly 

Just a few fixes to get the buttons working on the front end.