Fixed bug - meh