fix: add button type