bug fixing and adding tests to the button dropdown