Include one missing line +  a little fix...  Ty @DirtyJerz