Fixed bug i think