Fix button finding

Fix #325