Only same what I need.
A bit of code clean up.