Rebase fixes