fix
clean code