Fix bug to compile