Rebase fixes

Rebase fixes