Fixed it so that it works.