Add correct button placement for flutter and partner