Make sure the correct functions are exported