Fixing errors

Fixing compile errors