Fixade en bugg från senaste komitten
Implementerade en liten optimering där också