Bug fix: test