Fix button-click importing. Fixes #255