Fixes missing nullable annotations