Fix JS listener for update location button