Fixed dist build issue