Fix function test