Fix proof issues