updated for a potential bug

updated for a potential bug