add some code fix