Fix a minor bug in initializaion