Fixes error with incorrect heritage