Fixed an issue where the context would lose its owner and entity when its pushed.