Fix shit bug