just a code clean up