Chore: clean up