Small fix to test