Fix build

Temporary fixed for now.