clean code
fix last commit