Fix bug: shouldn't use self.dag. Broke the straddle unit test
Fix import error