[BUGFIX] fix include in tests