fix potential assetion error