Small fix on export