I solved this by using mount/swap-states as opposed to mount/swap. This means that the state var is still in the original NS but you can override the lifecycle methods
mount/swap-states
mount/swap