Fixed comilation errors from tests.