minor code clean