Trying fix on test func