Put the button back to see what happens.

Margins / Padding fixed in the last test. This might break it though...