Minor Fix | updated button type