Fix bug with compilation error