Fix the build after PR #9