fixed the big bug