code cleanup
fix merge conflict