Fixat några små buggar som blev av förra committen