Fixed error
Made it work