Changed test to use package