Fix buttons on import

Fixes #164