Add classes that two buttons appear on the right too not only one