fix: Clean code