Fixed bug real quick