Fixed my bug