fixes some code