fix: Make it work