Fixed bug, thx dave