How do you get rid of the keyboard button in one ui 2.0?

I'm using SwiftKey and since upgrading to one ui 2.0 there is a bar on the bottom of the keyboard that's taking up so much space. Tried to turn off the option in settings but it's greyed out. Anyone knows how to get rid of it? https://i.imgur.com/5kGDyah.png