Fixed build

Fixed build problem