Still trying to work out this momentum not found bug...