Fixed no test issue

Added test to class so it wouldn't fail.