fixed code error, hopefully