Fixed bug, i think..