Fixed failing query component test