fixes bug
adds this but it is not working