minor: code clean up