Small fix for compilation