Add some code fix