fix code, so test should run without errors