fix so it works