Small change so that it works, need to fix stack