Small fixes

Fixed the compile errors only.