Minor calculation fix