Changed it so that it should work. Needs testing