Fix error in 'test' code