update code to be lint correctly