fixed the damn bug