Fixed two missing things