Added a test to ensure this works on subclasses as well