Fixes optional type annotations.