Fixes incorrect class