fixed slight code error