fix copied code to fit