Fix rebake error
Merge pull request #978 from brianloveswords/fix-rebake-error

Fix rebake error