fixed bug in previous commit :)