Fixes silly bug