Fix recusion bug