quickly fixd some code