fixed compile errors with tests