Fixes wrong class reference