Fixed two possible bugs