Fix build

#18