Fix build

Temporary fix for now.