Minor change to forward inference test.