Fixed bug :/