fix a potential bug

This could cause a bug in the future.