Fixed jquery issue with button