Attempt to fix include error in test