Forgot to change class name to the new one (will break older version)