Fix for tests compilation