Fixed silly bug (thanks Nathan)