Fix error on build