fixed the pesky bug