fixed minor bug, everything seems to be working