Rebase error fixes