fixed compile errors :)