Fix error in previous version