Fix dist build