Fixed the bug - added id for button