Fix a bug

a fatal error