Fix test project

Fixed test project after #152