Fixed bug... sry!