Fixes bug in test.