Fuu, fix error caused by  variable :(
Fix small error. Script should be working nicely now :)