Ensure return types