Forgot the @ in class decorator ;)