Fix small build bug