updated bug when code was forked