Fix accidentally added class. Amends previous commit